tensor([[[[[0.5700, 1.0000],
           [1.0000, 0.0000]],

          [[0.3800, 0.3500],
           [0.5200, 0.5400]]],


         [[[0.0000, 0.3500],
           [0.0000, 0.0000]],

          [[0.3800, 0.3900],
           [0.3800, 0.3700]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.6100]],

          [[0.0000, 0.2900],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.2900],
           [0.5400, 0.4000]],

          [[0.2800, 0.2800],
           [0.2800, 0.2800]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.3200, 0.1500],
           [0.4800, 0.0000]]],


         [[[0.0000, 0.1500],
           [0.0000, 0.3100]],

          [[0.1200, 0.1100],
           [0.1200, 0.1200]]]],



        [[[[0.4300, 0.0000],
           [0.0000, 0.3900]],

          [[0.3100, 0.2100],
           [0.0000, 0.4600]]],


         [[[1.0000, 0.2100],
           [0.4600, 0.3000]],

          [[0.2200, 0.2200],
           [0.2300, 0.2200]]]]], grad_fn=<RoundBackward1>)